from langchain_openai import ChatOpenAI
from langchain.prompts import ChatPromptTemplate

# 配置 DeepSeek API（deepseek 兼容 OpenAI）
llm = ChatOpenAI(
    api_key = 'sk-XXXXXXXX',
    base_url = 'https://api.deepseek.com/v1',
    model='deepseek-chat'# 或其他 DeepSeek 模型
)

# # 直接调用
# response = llm.invoke("你好，DeepSeek！")
# print(response.content)
#
# # 使用 PromptTemplate
# prompt = ChatPromptTemplate.from_messages([
#     ("system", "你是一个专业的AI助手"),
#     ("user", "{question}")
# ])
# chain = prompt | llm
# result = chain.invoke({"question": '请解释机器学习的基本概念'})
# print(result.content)

prompt = """
请生成工具调用请求（JSON格式），包含以下字段：
- "tool_name": 工具名称（如 "get_weather"）
- "tool_args": 工具参数（如 {"location": "北京"}）

用户问题：查询大连天气
"""
response = llm.invoke(prompt)
print(response)